import time

import pymysql
import os


def rows(self):
    # 数据库连接对象
    conn = pymysql.connect(host='localhost', port=3306, user='root', password="root", db="tab")
    # 游标对象
    cur = conn.cursor()
    # 查询的sql语句
    sql = "SELECT * FROM tab"
    cur.execute(sql)
    # 获取查询到的数据, 是以二维元组的形式存储的, 所以读取需要使用 data[i][j] 下标定位
    data = cur.fetchall()
    x = 0
    for i in data:
        y = 0
        for j in i:
            # self.tablewidget.setItem(x, y, QtWidgets.QTableWidgetItem(str(data[x][y])))
            y = y + 1
        x = x + 1

    cur.close()  # 游标关闭
    conn.close()  # 数据库关闭

    return x, data
def Image_Save_Dir(Now_Time):
    Parent_Directory = 'picture'
    Save_Dir = "./picture/" + Now_Time

    # print("Save_Dir name is ", Save_Dir)
    if not os.path.exists(Parent_Directory):
        print("Parent_Directory path is not exist.")
        os.mkdir('picture')

    if not os.path.exists(Save_Dir):
        print("Given file path is not exist.")
        os.mkdir(Save_Dir)
def Number_Of_Picture():
    Time=time.strftime("%Y-%m-%d",time.localtime(time.time()))
    # 获取到当前文件夹下面所有的文件
    Path="./picture/" + Time

    Image_Save_Dir(Time)

    '''
    os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。
    这个列表以字母顺序。 
    它不包括 . 和 .. 即使它在文件夹中。
    只支持在 Unix, Windows 下使用。
    '''
    all_files = os.listdir(Path)  # os.curdir 表示当前目录 curdir:currentdirectory
    # dict() 函数用于创建一个字典
    type_dict = dict()

    for each_file in all_files:
        # 如果说我们的each_file是文件夹
        if os.path.isdir(each_file):
            type_dict.setdefault("文件夹", 0)
            type_dict["文件夹"] += 1  #计数
        else:
            # 如果不是文件夹，而是文件，统计我们的文件

            ext = os.path.splitext(each_file)[1]  # 获取到文件的后缀
            type_dict.setdefault(ext, 0)      # 分割文件名与路径
            type_dict[ext] += 1
    try:
        number_of_picture=type_dict[('.jpg')]
    except Exception as e:
        print(e)
        number_of_picture=0
    # for each_type in type_dict:
    #     print("该文件夹下面有类型为{}的文件{}个".format(each_type, type_dict[each_type]))
    # print(number_of_picture)
    return [str(number_of_picture),str(Path)]

    # return number_of_picture
if __name__ == "__main__":
    path_of_folder = 'D:/图片/白老板4k'
    n=Number_Of_Picture()
    # print(n)